import matplotlib.pyplot as plt

from random_walk import RandomWork

flags = 1
while flags:
    rw = RandomWork(5000)
    rw.fill_work()
    plt.figure(dpi=50, figsize=(16, 9))
    plt.scatter(rw.x_values, rw.y_values, cmap=plt.cm.Blues, s=1)
    plt.axes().get_xaxis().set_visible(False)
    plt.axes().get_yaxis().set_visible(False)
    plt.show()

    stop = input("终止条件(T)")
    if stop == 'T':
        flags = 0
